import json
from pytdx.hq import TdxHq_API
from pytdx.params import TDXParams


def block():
    api = TdxHq_API(raise_exception=False)
    file = open('data.txt', mode='w', encoding='utf-8')
    with api.connect('110.41.147.114', 7709):
        data = api.get_and_parse_block_info(TDXParams.BLOCK_DEFAULT)
        for d in data:
            s = "%s %d %d %s\n" % (d['blockname'], d['block_type'], d['code_index'], d['code'])
            file.write(s)
    file.close()


def quotes():
    api = TdxHq_API(raise_exception=False)
    with api.connect('8.129.13.54', 7709):
        list = [(0, '002680'), (0, '002619'), (0, '002618'), (0, '002615'), (0, '002611'), (0, '002608'), (0, '002604'), (0, '002594'), (0, '002587'), (0, '002560'), (0, '002522'), (0, '002515'), (0, '002509'), (0, '002501'), (0, '002477'), (0, '002473'), (0, '002471'), (0, '002465'), (0, '002464'), (0, '002452'), (0, '002450'), (0, '002447'), (0, '002427'), (0, '002413'), (0, '002406'), (0, '002361'), (0, '002359'), (0, '002345'), (0, '002341'), (0, '002325'), (0, '002310'), (0, '002297'), (0, '002278'), (0, '002260'), (0, '002224'), (0, '002220'), (0, '002208'), (0, '002204'), (0, '002147'), (0, '002146'), (0, '002143'), (0, '002132'), (0, '002131'), (0, '002122'), (0, '002106'), (0, '002099'), (0, '002097'), (0, '002089'), (0, '002078'), (0, '002073'), (0, '002071'), (0, '002070'), (0, '002069'), (0, '002044'), (0, '002018'), (0, '002015'), (0, '001979'), (0, '001301'), (0, '001223'), (0, '000982'), (0, '000981'), (0, '000979'), (0, '000975'), (0, '000956'), (0, '000939'), (0, '000936'), (0, '000926'), (0, '000916'), (0, '000908'), (0, '000866'), (0, '000835'), (0, '000832'), (0, '000827'), (0, '000817'), (0, '000816'), (0, '000809'), (0, '000805'), (0, '000795'), (0, '000787'), (0, '000786'), (0, '000783'), (0, '000780'), (0, '000778'), (0, '000769'), (0, '000765'), (0, '000763'), (0, '000760'), (0, '000748'), (0, '000730'), (0, '000722'), (0, '000699'), (0, '000693'), (0, '000690'), (0, '000689'), (0, '000687'), (0, '000675'), (0, '000673'), (0, '000671'), (0, '000669'), (0, '000667'), (0, '000662'), (0, '000660'), (0, '000658'), (0, '000653'), (0, '000652'), (0, '000621'), (0, '000618'), (0, '000613'), (0, '000611'), (0, '000602'), (0, '000594'), (0, '000588'), (0, '000585'), (0, '000583'), (0, '000578'), (0, '000572'), (0, '000571'), (0, '000569'), (0, '000562'), (0, '000556'), (0, '000552'), (0, '000549'), (0, '000542'), (0, '000540'), (0, '000536'), (0, '000535'), (0, '000527'), (0, '000522'), (0, '000515'), (0, '000511'), (0, '000508'), (0, '000502'), (0, '000418'), (0, '000415'), (0, '000413'), (0, '000412'), (0, '000406'), (0, '000405'), (0, '000096'), (0, '000090'), (0, '000055'), (0, '000047'), (0, '000046'), (0, '000033'), (0, '000027'), (0, '000024'), (0, '000018'), (0, '000015'), (0, '000013'), (0, '000003'), (0, '300763'), (0, '300833'), (1, '600563')]
        list = list[:2]
        data = api.get_security_quotes([(0, '002604')])
        if data:
            for d in data:
                j = json.dumps(d)
                print(j)


if __name__ == '__main__':
    quotes()